package demo4.mapper;

import demo4.pojo.User;
import demo4.util.IDEBUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

@Component
public class UserMapper implements IUserMapper{
    @Autowired
    private IDEBUtil dbUtil;
    public int insert(User user) {
       // System.out.println("UserMapper:向user表中插入用户数据");
        try (Connection connection = dbUtil.getConnection();){
            String sal="INSERT INTO user(username,password,nickname,age )"+
                       " VALUES(?,?,?,?)";
            PreparedStatement ps=connection.prepareStatement(sal);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNickname());
            ps.setInt(4,user.getAge());
            return ps.executeUpdate();

        } catch (SQLException e) {
           e.printStackTrace();
        }
        /*System.out.println("UserMapper:执行Insert操作");
        System.out.println("UserMapper:执行完毕，返回影响条目数");*/
        return 0;
    }

    @Override
    public User selectByUsername(String username) {
        try (Connection connection=dbUtil.getConnection();){
            String sql="SELECT id,username,password,nickname,age "+
                    " FROM user "+
                    "WHERE username=?";

            PreparedStatement ps=connection.prepareStatement(sql);
            ps.setString(1,username);
            ResultSet rs= ps.executeQuery();

            if(rs.next()){
                User user=new User();
                user.setId(rs.getInt("id"));
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                return user;
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return null;
    }

    @Override
    public List<User> selectAll() {
        List<User> userList=new ArrayList<>();
        try (Connection connection=dbUtil.getConnection();){
            String sql="SELECT id,username,password,nickname,age "+
                    " FROM user ";
            Statement st=connection.createStatement();
            ResultSet rs= st.executeQuery(sql);
            while(rs.next()){
                User user=new User();
                user.setId(rs.getInt("id"));
                user.setUsername(rs.getString("username"));
                user.setPassword(rs.getString("password"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                userList.add(user);
            }

        } catch (SQLException e) {
            e.printStackTrace();
        }
        return userList;
    }

    @Override
    public int update(User user) {
        try (Connection connection=dbUtil.getConnection();){
            String sql="UPDATE user" +
                    " SET nickname=?,password=?,age=?" +
                    " WHERE username=?";
            PreparedStatement ps=connection.prepareStatement(sql);
            ps.setString(1,user.getNickname());
            ps.setString(2,user.getPassword());
            ps.setInt(3,user.getAge());
            ps.setString(4,user.getUsername());
            return ps.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return 0;
    }
}
